package com.sqpoll.android.widget;

import android.content.Context;
import android.util.AttributeSet;
import android.view.View;
import android.view.ViewGroup;
import android.widget.ListAdapter;
import android.widget.ListView;

public class ExpandableHeightListView extends ListView {

	boolean expanded = false;
	boolean maxExpanded = false;

	public ExpandableHeightListView(Context context) {
		super(context);
	}

	public ExpandableHeightListView(Context context, AttributeSet attrs) {
		super(context, attrs);
	}

	public ExpandableHeightListView(Context context, AttributeSet attrs, int defStyle) {
		super(context, attrs, defStyle);
	}

	public boolean isExpanded() {
		return expanded;
	}

	@Override
	public void onMeasure(int widthMeasureSpec, int heightMeasureSpec) {
		// HACK! TAKE THAT ANDROID!
		if (isExpanded()) {
			if (maxExpanded) {
				// Calculate entire height by providing a very large height hint.
				// But do not use the highest 2 bits of this integer; those are
				// reserved for the MeasureSpec mode.
				int expandSpec = MeasureSpec.makeMeasureSpec(Integer.MAX_VALUE >> 2, MeasureSpec.AT_MOST);
				super.onMeasure(widthMeasureSpec, expandSpec);
	
				ViewGroup.LayoutParams params = getLayoutParams();
				params.height = getMeasuredHeight();
			}
			else {
				ListAdapter listAdapter = getAdapter();
				if (listAdapter == null) {
					return;
				}
				
				int totalHeight = 0;
				int desiredWidth = MeasureSpec.makeMeasureSpec(getWidth(), MeasureSpec.AT_MOST);

				for (int i = 0; i < listAdapter.getCount(); i++) {
					View listItem = listAdapter.getView(i, null, this);
					listItem.measure(desiredWidth, MeasureSpec.UNSPECIFIED);
					totalHeight += listItem.getMeasuredHeight();
				}
				
				//int expandSpec = MeasureSpec.makeMeasureSpec(Integer.MAX_VALUE >> 2, MeasureSpec.AT_MOST);
				int expandSpec = MeasureSpec.makeMeasureSpec(totalHeight, MeasureSpec.AT_MOST);
				super.onMeasure(widthMeasureSpec, expandSpec);
				
				ViewGroup.LayoutParams params = getLayoutParams();
				params.height = totalHeight + (getDividerHeight() * (listAdapter.getCount() - 1));
				
				//int expandSpec = MeasureSpec.makeMeasureSpec(Integer.MAX_VALUE >> 2, MeasureSpec.AT_MOST);
				//int expandSpec = MeasureSpec.makeMeasureSpec(params.height, MeasureSpec.AT_MOST);
				//super.onMeasure(widthMeasureSpec, expandSpec);	
			}			
		} else {
			super.onMeasure(widthMeasureSpec, heightMeasureSpec);
		}
	}

	public void setExpanded(boolean expanded) {
		this.expanded = expanded;
		this.maxExpanded = true;
	}
	
	public void setExpanded(boolean expanded, boolean maxExpanded) {
		this.expanded = expanded;
		this.maxExpanded = maxExpanded;
	}
}
